package info.wangyuan.agent.entity.dto;

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serial;
import java.io.Serializable;
import java.util.List;

/**
 * Rerank请求参数
 *
 * @author Albert
 * @since 2025-09-07 16:38:22
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class RerankRequestDTO implements Serializable {

    @Serial
    private static final long serialVersionUID = -1272322443949808505L;

    private String model;

    private String query;

    private List<String> documents;

    @JsonProperty("return_documents")
    private boolean returnDocuments = false;

    @JsonProperty("max_chunks_per_doc")
    private Integer maxChucksPerDoc = 10;

    @JsonProperty("overlap_tokens")
    private Integer overlapTokens = 80;

}
